const mongoose = require('mongoose');
const dotenv = require('dotenv');
const User = require('./models/User');

// 加载环境变量
dotenv.config();

// MongoDB连接
mongoose.connect(process.env.MONGO_URI)
  .then(() => {
    console.log('MongoDB连接成功');
    initDefaultUser();
  })
  .catch((err) => {
    console.error('MongoDB连接失败:', err);
    process.exit(1);
  });

// 创建默认用户
const initDefaultUser = async () => {
  try {
    // 检查默认用户是否已存在
    const existingUser = await User.findOne({ account: 'treasure' });
    
    if (existingUser) {
      console.log('默认用户已存在');
      mongoose.connection.close();
      return;
    }
    
    // 创建默认用户
    const defaultUser = new User({
      username: '管理员',
      account: 'treasure',
      password: '123456',
      email: 'admin@example.com',
      phone: '13800138000',
      role: 'admin',
      avatar: 'admin-avatar.png',
      status: 1,
      createdAt: Date.now(),
      updatedAt: Date.now()
    });
    
    await defaultUser.save();
    console.log('默认用户创建成功：');
    console.log('- 账号：treasure');
    console.log('- 密码：123456');
    
    mongoose.connection.close();
  } catch (error) {
    console.error('创建默认用户失败:', error);
    mongoose.connection.close();
  }
};